<?php

/**
 * Tree 树型类(无限分类)
 * 通用的树形类
 * 
 */
class Tree {

    public $arr = array();
    //生成树形结构所需的符号，可以换成图片
    public $icon = array('│', '├', '└');

    //构造函数，初始化类
    function __construct($arr = array()) {
        $this->arr = $arr;
        $this->ret = '';
        return is_array($arr);
    }

    //得到父级数组
    function get_parent($myid) {
        $newarr = array();
        if (!isset($this->arr[$myid])) {
            return false;
        }
        $fid = $this->arr[$myid]['catePID'];
        $fid = $this->arr[$fid]['catePID'];
        if (is_array($this->arr)) {
            foreach ($this->arr as $id => $a) {
                if ($a['catePID'] == $fid)
                    $newarr[$id] = $a;
            }
        }
        return $newarr;
    }

    //得到子级数组
    function get_child($myid) {
        $a = $newarr = array();
        if (is_array($this->arr)) {
            foreach ($this->arr as $id => $a) {
                if ($a['catePID'] == $myid)
                    $newarr[$id] = $a;
            }
        }
        return $newarr ? $newarr : false;
    }

    //得到当前位置数组
    function get_pos($myid, &$newarr) {
        $a = array();
        if (!isset($this->arr[$myid]))
            return false;
        $newarr[] = $this->arr[$myid];
        $fid = $this->arr[$myid]['catePID'];
        if (isset($this->arr[$fid])) {
            $this->get_pos($fid, $newarr);
        }
        if (is_array($newarr)) {
            krsort($newarr);
            foreach ($newarr as $v) {
                $a[$v['id']] = $v;
            }
        }
        return $a;
    }

    /**
     * -------------------------------------  
     *  得到树型结构  
     * -------------------------------------  
     * @param $myid 表示获得这个ID下的所有子级  
     * @param $str 生成树型结构的基本代码，例如："<option value=\$id \$selected>\$spacer\$name</option>"   
     * @param $sid 被选中的ID, 比如在做树形下拉框的时候需要用到  
     * @param $adds  
     * @param $str_group  
     */
    function get_tree($myid, $str, $sid = 0, $adds = '', $str_group = '') {
        $number = 1;
        $child = $this->get_child($myid);
        if (is_array($child)) {
            $total = count($child);
            foreach ($child as $id => $a) {
                $j = $k = '';
                if ($number == $total) {
                    $j .= $this->icon[2];
                } else {
                    $j .= $this->icon[1];
                    $k = $adds ? $this->icon[0] : '';
                }
                $spacer = $adds ? $adds . $j : '';
                $cid=$a['id'];
                $selected = $cid == $sid ? 'selected' : '';
                @extract($a);
                @$fid == 0 && $str_group ? eval("\$nstr = \"$str_group\";") : eval("\$nstr = \"$str\";");
                $this->ret .= $nstr;
                $this->get_tree($id, $str, $sid, $adds . $k . '&nbsp;', $str_group);
                $number++;
            }
        }
        return $this->ret;
    }

    /**
     * 同上一方法类似,但允许多选  
     */
    function get_tree_multi($myid, $str, $sid = 0, $adds = '') {
        $number = 1;
        $child = $this->get_child($myid);
        if (is_array($child)) {
            $total = count($child);
            foreach ($child as $id => $a) {
                $j = $k = '';
                if ($number == $total) {
                    $j .= $this->icon[2];
                } else {
                    $j .= $this->icon[1];
                    $k = $adds ? $this->icon[0] : '';
                }
                $spacer = $adds ? $adds . $j : '';
                $selected = $this->have($sid, $id) ? 'selected' : '';
                @extract($a);
                eval("\$nstr = \"$str\";");
                $this->ret .= $nstr;
                $this->get_tree_multi($id, $str, $sid, $adds . $k . '&nbsp;');
                $number++;
            }
        }
        return $this->ret;
    }

    function have($list, $item) {
        return(strpos(',,' . $list . ',', ',' . $item . ','));
    }

    /**
     * 格式化数组  
     */
    function getArray($myid = 0, $sid = 0, $adds = '') {
        $number = 1;
        $child = $this->get_child($myid);
        if (is_array($child)) {
            $total = count($child);
            foreach ($child as $id => $a) {
                $j = $k = '';
                if ($number == $total) {
                    $j .= $this->icon[2];
                } else {
                    $j .= $this->icon[1];
                    $k = $adds ? $this->icon[0] : '';
                }
                $spacer = $adds ? $adds . $j : '';
                @extract($a);
                $a['cateName'] = $spacer . ' ' . $a['cateName'];
                $this->ret[$a['id']] = $a;
                $fd = $adds . $k . '&nbsp;';
                $this->getArray($id, $sid, $fd);
                $number++;
            }
        }
        return $this->ret;
    }

}

?>
